package com.macro.mall.tiny.common.example;

import javax.crypto.Cipher;
import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.PrivateKey;
import java.security.PublicKey;
import java.util.Base64;

/**
 * @author: i_kun
 * @Desc: o_。
 * @create: 2025-06-25 16:46
 */

public class RSAExample {
    public static void main(String[] args) throws Exception {
        // 原始待加密文本
        String plainText = "Hello, RSA!";

        // 1. 生成RSA密钥对（2048位）
        // 获取RSA密钥对生成器实例
        KeyPairGenerator keyGen = KeyPairGenerator.getInstance("RSA");
        // 初始化密钥长度。2048位是当前安全标准，低于1024位被认为不安全[1,3](@ref)
        keyGen.initialize(2048);
        // 生成密钥对（包含公钥和私钥）
        KeyPair keyPair = keyGen.generateKeyPair();
        // 获取公钥（用于数据加密）
        PublicKey publicKey = keyPair.getPublic();
        // 获取私钥（必须保密，用于解密）
        PrivateKey privateKey = keyPair.getPrivate();

        // 2. 使用公钥加密 // 调用加密方法
        byte[] encryptedData = encrypt(plainText, publicKey);
        // 将二进制密文转换为Base64字符串输出
        System.out.println("加密结果 (Base64): " + Base64.getEncoder().encodeToString(encryptedData));

        // 3. 使用私钥解密 // 调用解密方法
        String decryptedText = decrypt(encryptedData, privateKey);

        // 输出解密后的原始文本
        System.out.println("解密结果: " + decryptedText);
    }

    /**
     * RSA加密方法
     * @param plainText 明文文本
     * @param publicKey 公钥对象
     * @return 加密后的字节数组
     */
    public static byte[] encrypt(String plainText, PublicKey publicKey) throws Exception {
        // 获取RSA密码器实例（默认使用ECB模式，PKCS1Padding填充）
        Cipher cipher = Cipher.getInstance("RSA");
        // 初始化为加密模式，并注入公钥
        cipher.init(Cipher.ENCRYPT_MODE, publicKey);
        // 执行加密操作
        return cipher.doFinal(plainText.getBytes("UTF-8"));
        // 注意：RSA单次加密数据长度受限（2048位密钥最大加密117字节）[5](@ref)
        // 长文本需分段加密（此处未实现）
    }

    /**
     * RSA解密方法
     * @param encryptedData 加密后的字节数组
     * @param privateKey 私钥对象
     * @return 解密后的原始字符串
     */
    public static String decrypt(byte[] encryptedData, PrivateKey privateKey) throws Exception {
        Cipher cipher = Cipher.getInstance("RSA"); // 获取RSA密码器实例
        cipher.init(Cipher.DECRYPT_MODE, privateKey); // 初始化为解密模式，并注入私钥
        byte[] decryptedBytes = cipher.doFinal(encryptedData); // 执行解密操作
        return new String(decryptedBytes, "UTF-8"); // 将解密后的字节数组转为UTF-8字符串
    }
}
